/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.wsinfra.apps;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Properties;
import java.util.StringTokenizer;

import javax.xml.namespace.QName;
import javax.xml.soap.SOAPMessage;

import org.apache.xmlbeans.XmlCursor;

import cgl.narada.wsinfra.deployment.Filter;
import cgl.narada.wsinfra.deployment.FilterPipeline;
import cgl.narada.wsinfra.deployment.FilterPipelineFactory;
import cgl.narada.wsinfra.deployment.NetworkSubstrate;
import cgl.narada.wsinfra.deployment.ServiceMessageListener;
import cgl.narada.wsinfra.deployment.impl.NBNetworkSubstrate;
import cgl.narada.wsinfra.exception.DeploymentException;
import cgl.narada.wsinfra.exception.MessageFlowException;
import cgl.narada.wsinfra.exception.ProcessingException;
import cgl.narada.wsinfra.schemas.addressing.EndpointReferenceType;
import cgl.narada.wsinfra.schemas.addressing.FromDocument;
import cgl.narada.wsinfra.schemas.soap.EnvelopeDocument;
import cgl.narada.wsinfra.util.SoapEnvelopeConversion;
import cgl.narada.wsinfra.util.SoapMessageAlteration;
import cgl.narada.wsinfra.util.SoapPrinter;
import cgl.narada.wsinfra.wsa.WsaEnvelopeCreator;
import cgl.narada.wsinfra.wsa.WsaEprCreator;
import cgl.narada.wsinfra.wsa.WsaProcessingFactory;
import cgl.narada.wsinfra.wse.WseQNames;

public class WseSourceEndpoint implements ServiceMessageListener {
  private NetworkSubstrate networkSubstrate;
  private FilterPipelineFactory filterPipelineFactory;
  private FilterPipeline filterPipeline;
  private WsaEprCreator wsaEprCreator;
  private WsaEnvelopeCreator wsaEnvelopeCreator;
  private SoapEnvelopeConversion soapEnvelopeConversion;
  private SoapMessageAlteration soapMessageAlteration;
  private Filter loggingFilter;
  private Filter counterFilter;
  private Filter wseSourceFilter;
  private String sourceAddress;
  private int tracker;
  private WseQNames wseQNames;
  private String moduleName = "WseSourceEndpoint: ";

  public WseSourceEndpoint(String sourceAddress, 
			   String subManagerAddress,
			   Properties props) 
    throws DeploymentException {
    this.sourceAddress = sourceAddress;
    
    filterPipelineFactory =  FilterPipelineFactory.getInstance();
    filterPipeline = filterPipelineFactory.newFilterPipeline("WsePipeline");
    
    wseSourceFilter = new WseSourceFilter(sourceAddress, subManagerAddress);
    loggingFilter = new LoggingFilter();
    counterFilter = new CounterFilter();

    filterPipeline.addFilter(wseSourceFilter);
    filterPipeline.addFilter(counterFilter);
    filterPipeline.addFilter(loggingFilter);

    String pipelineInfo = moduleName + "Configured FilterPipeline" + 
      "\n\tWseSourceFilter=" + wseSourceFilter.getPositionInFilterPipeline() +
      "\n\tCounterFilter=" +  counterFilter.getPositionInFilterPipeline() +
      "\n\tLoggingFilter=" + loggingFilter.getPositionInFilterPipeline()+ "\n";

    System.out.println( pipelineInfo );
    
    networkSubstrate = new NBNetworkSubstrate();
    networkSubstrate.configureNetworking(props);
    
    networkSubstrate.registerPipeline(filterPipeline);
    filterPipeline.registerNetworkSubstrate(networkSubstrate);

    wsaEprCreator = WsaProcessingFactory.getWsaEprCreator();
    wsaEnvelopeCreator = WsaProcessingFactory.getWsaEnvelopeCreator();

    soapEnvelopeConversion = SoapEnvelopeConversion.getInstance();
    soapMessageAlteration = SoapMessageAlteration.getInstance();

    wseQNames = WseQNames.getInstance();
  }

  public void onMessage(SOAPMessage soapMessage) {
    System.out.println(moduleName + "Received SOAP Message from the " +
		       "underlying infrastructure");
    printSoapMessage(soapMessage);
  }

  public void onException(Exception exception) {


  }

  public void changeFilterOrder() throws DeploymentException {
    int position = loggingFilter.getPositionInFilterPipeline() ;
    if (position == 0) {
      filterPipeline.moveFilter(loggingFilter, 1);
    } 

    if (position == 1) {
      filterPipeline.moveFilter(loggingFilter, 0);
    } 
    
  }

  public void printSoapMessage(SOAPMessage soapMessage) {
    String soapMsg = null;
    try {
      soapMsg = SoapPrinter.getStringRepresentation(soapMessage);
    } catch (Exception e) {
      System.out.println(moduleName + "Problems printing SOAPMessage"); 
    }
    System.out.println(soapMsg);
  }

  public String createLinkTo(Properties props, String linkType) 
    throws DeploymentException { 
    if (networkSubstrate == null) {
      System.out.println(moduleName + "The network substrate is null");
    }
    String linkId = 
      ( (NBNetworkSubstrate)networkSubstrate).createLinkTo(props, linkType);
    return linkId;
  }


  public void sendSOAPMessage(SOAPMessage soapMessage) 
    throws DeploymentException, MessageFlowException, ProcessingException {
    filterPipeline.processMessageFromApplication(soapMessage); 
  }


  private EnvelopeDocument createEnvelopeDocument() {
    EndpointReferenceType sourceEpr = wsaEprCreator.createEpr(sourceAddress);
    FromDocument from = FromDocument.Factory.newInstance();
    from.setFrom(sourceEpr);
    EnvelopeDocument envelopeDocument = 
      wsaEnvelopeCreator.createSoapEnvelope(from);
    
    QName qname = new QName("http://www.naradabrokering.org",
			    "Application-Content");
    XmlCursor bodyCursor = 
      envelopeDocument.getEnvelope().getBody().newCursor();
    tracker++;
    String value = "Tracker :" + tracker + "\n" +
      "The quick brown fox jumps upon the lazy dog";
    bodyCursor.toNextToken();

    String catUri = "http://www.naradabrokering.org/catalog";
    bodyCursor.insertElement("catalog", catUri);
    bodyCursor.toPrevToken();
    bodyCursor.insertElement("cd", catUri);
    bodyCursor.toPrevToken();
    bodyCursor.insertElementWithText("price", catUri, "11.90");

    String xpathString = "<catalog> <cd> <title>Hide your heart</title> <artist>Bonnie Tyler</artist>  <country>UK</country>  <company>CBS Records</company> <price>11.90</price>  <year>1988</year>  </cd>  </catalog>";
    //bodyCursor.insertChars(xpathString);
    bodyCursor.insertElementWithText(qname, value);
    bodyCursor.dispose();


    bodyCursor = envelopeDocument.getEnvelope().getBody().newCursor();
    String poUri = "http://openuri.org/easypo";
    bodyCursor.insertElement("purchase-order", poUri);
    bodyCursor.toPrevToken();
    bodyCursor.insertElement("line-item", poUri);
    
    addTopicInfo(envelopeDocument);
    bodyCursor.toPrevToken();
    bodyCursor.insertElementWithText("price", poUri, "300");
    bodyCursor.dispose();

    return envelopeDocument;
  }
    
  public SOAPMessage createSOAPMessage() {
    EnvelopeDocument envelopeDocument = createEnvelopeDocument();
    /*
      EnvelopeDocument envelopeCopy=(EnvelopeDocument)envelopeDocument.copy();
      System.out.println(moduleName + "COPY OF THE EnvelopeDocument is \n" +
      envelopeCopy.toString());
    */

    SOAPMessage soapMessage = null;
    try {
      soapMessage =  soapEnvelopeConversion.getSOAPMessage(envelopeDocument);
    } catch (ProcessingException ex) {
      String errorReport ="Problems converting to javax.xml.SOAPMessage\n"
	+ ex.toString();
      System.out.println(errorReport);
    }
    
    return soapMessage;
  }


  
  public void evaluateQuery() {
    String nsText = "declare namespace po = 'http://openuri.org/easypo';";
    String pathText = "$this//po:purchase-order/po:line-item";
    
    EnvelopeDocument envDocument = createEnvelopeDocument();
    String xqueryFilterConstraint = nsText + pathText;
    XmlCursor xqueryCursor = envDocument.newCursor();
    XmlCursor resultCursor = xqueryCursor.execQuery(xqueryFilterConstraint);
    
    String result = resultCursor.getTextValue();
    
    if (result != null) {
      System.out.println(moduleName + "Matching success!");
    } else {
      System.out.println(moduleName + "Matching failed!");
    }
    xqueryCursor.dispose();
    resultCursor.dispose();
    
    
    String xpathFilterConstraint = "declare namespace c1='http://www.naradabrokering.org/catalog'" + "$this//c1:catalog/c1:cd/c1:price";
    
    EnvelopeDocument envelopeDocument = createEnvelopeDocument();
    System.out.println(moduleName + "\n" + envelopeDocument);
    XmlCursor xmlCursor = envelopeDocument.newCursor();
    xmlCursor.selectPath(xpathFilterConstraint);
    
    boolean matchingSuccess = false;
    int selectionCount = xmlCursor.getSelectionCount();
    
    
    if (selectionCount != 0) {
      matchingSuccess = true;
      System.out.println(moduleName + "Success in matching.");
    }
    
    xmlCursor.dispose();

  }

  private void addTopicInfo(EnvelopeDocument envelopeDocument) {
    String topicName = "/Literature/Shakespere";
    QName qName = wseQNames.getTopicQNameNBX();
    boolean added = soapMessageAlteration.addToSoapHeader(envelopeDocument, 
							  qName, topicName);

    if (!added) {
      System.out.println(moduleName + "Addition of Topic info NOT Successful");
    }
  }

  public static void main(String[] args) {
    if (args.length != 3) {
      String errorReport = 
	"Usage: java cgl.narada.wsinfra.apps.WseSourceEndpoint " +
	"[hostname] [TCPServerPort] [WseSubManagerAddress]";
      System.out.println(errorReport);
      System.exit(0);
    }

    String hostname = args[0];
    String port = args[1];
    String wseSubManagerAddress = args[2];
    Properties props = new Properties();
    props.put("TCPServerPort", port);
    
    BufferedReader r = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer tok;
    String cmd;

    try {
      String sourceAddress = "tcp://" + hostname + ":" + port;
      WseSourceEndpoint endpoint = 
	new WseSourceEndpoint(sourceAddress, wseSubManagerAddress, props);
      System.out.println("Listening to TCP communications on port [" + port 
			 + "]. \n Type h for help on commands");
      while (true) {
      	cmd = r.readLine();
	if (cmd == null) {
	  System.out.println("Terminating program ....");
	  break;
	}
	if(cmd.startsWith("h")) {
	  System.out.println("Create Link           -> cr hostname portNum ");
	  System.out.println("Send Data [from] [to] -> s  [source] [sink]" );
	}
	
	if (cmd.startsWith("cr")) {
	  tok = new StringTokenizer(cmd, " ");
	  tok.nextToken(); /* skipping the token "cr" */
	  Properties linkProps = new Properties();
	  linkProps.put("hostname", tok.nextToken());
	  linkProps.put("portnum", tok.nextToken());
	  String _linkId = endpoint.createLinkTo(linkProps, "tcp");
	  System.out.println("Created link with id => " + _linkId); 
	} /** end create link */
	
	if (cmd.startsWith("s")) {
	  SOAPMessage soapMessage = endpoint.createSOAPMessage();
	  endpoint.sendSOAPMessage(soapMessage);
	} 

	if (cmd.startsWith("e")) {
	  endpoint.evaluateQuery();
	}
	
      } /* end while(true) */

    }catch (Exception ioe) {
      ioe.printStackTrace();
    } /* end try-catch */
  }/*end main() */

}
